# Directory containing preprocessed MSCOCO data.
MSCOCO_DIR="$PWD/im2txt/data/mscoco"

# Inception v3 checkpoint file.
INCEPTION_CHECKPOINT="$PWD/im2txt/data/inception_v3.ckpt"

# Directory to save the model.
MODEL_DIR="$PWD/im2txt/model"

# Build the model.
#cd research/im2txt
bazel build -c opt //im2txt/...

# Run the training script.
bazel-bin/im2txt/train \
--input_file_pattern="${MSCOCO_DIR}/train-?????-of-00256" \
--inception_checkpoint_file="${INCEPTION_CHECKPOINT}" \
--train_dir="${MODEL_DIR}/train" \
--train_inception=false \
--number_of_steps=1000000
